作者:最美的你 | 来源:互联网 | 2023-08-27 12:48
Tensorflow | TF . data . dataset . from _ tensor _ slices()
原文:https://www . geesforgeks . org/tensorflow-TF-data-dataset-from _ tensor _ slices/
借助**tf.data.Dataset.from_tensor_slices()**
方法,我们可以用tf.data.Dataset.from_tensor_slices()
方法得到物体形式的数组切片。
语法: tf.data.Dataset.from_tensor_slices(list)
返回:返回切片元素的对象。
示例#1 :
在这个示例中,我们可以看到,通过使用tf.data.Dataset.from_tensor_slices()
方法,我们能够获得列表或数组的切片。
# import tensorflow
import tensorflow as tf
# using tf.data.Dataset.from_tensor_slices() method
gfg = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4, 5])
for ele in gfg:
print(ele.numpy())
输出:
1
2
3
4
5
例 2 :
# import tensorflow
import tensorflow as tf
# using tf.data.Dataset.from_tensor_slices() method
gfg = tf.data.Dataset.from_tensor_slices([[5, 10], [3, 6]])
for ele in gfg:
print(ele.numpy())
输出:
[5,10]
【3,6】